// util/redisUtil.go

package util

import (
	"context"
	"github.com/go-redis/redis/v8"
	"time"
)

var rdb *redis.Client

func init() {
	// 初始化 Redis 客户端
	rdb = redis.NewClient(&redis.Options{
		Addr:     "localhost:6379", // 请根据实际情况修改为正确的 Redis 地址
		Password: "",               // 如果有密码，请填写 Redis 密码，否则留空
		DB:       0,
	})
}

// GetToken 从 Redis 中获取 Token
func GetToken(token string) (string, error) {
	return rdb.Get(context.Background(), token).Result()
}

// SetToken 将 Token 存储到 Redis 中
func SetToken(token string, expiration time.Duration) error {
	return rdb.Set(context.Background(), token, token, expiration).Err()
}
